Inject a Turn, Not a Token: Learning Where to Redirect LLM Reasoning for Self-Correction
学习在何处注入新用户轮次以重定向 LLM 推理自纠正
🧭 LLM 推理自纠正 · 干预位置学习
KAIST
UPTURN 用 130K 参数头学习失败推理的干预位置,以新用户轮次注入纠正指令,无需微调即提升数学推理准确率
📌 工作参考价值(4 点)
- 在对话式推荐/搜索 Agent 的 multi-step 推理或工具调用失败时,可训练轻量 MLP 头(冻结 encoder)从若干候选 step 中选一个「重试起点」,只保留前缀、重新生成后缀,避免整轮重跑;该头只以最终任务成功(如是否下单/点击)做弱监督,不需要过程标注。
- 干预时把纠错指令作为新的 user turn 注入,而不是拼在原生成文本里 inline 注入;在固定生成预算下,角色转换更容易改变模型后续生成分布,这对包含 system/assistant/user 轮次的搜索推荐多轮对话尤其适用。
- 该框架与现有 RL(如 GRPO)和 self-correction 正交,可在强化学习后的 policy 上叠加,作为推理时纠错模块,适合业务中不重训大模型、只加小模型控制干预时机的低风险上线方式。
- 但注意其训练信号只有 final-answer correctness,对没有明确二元成功信号的推荐场景(如点击/转化稀疏)需要重新设计 reward,迁移价值主要在「何时重试」的选择器+「新 user turn」重定向技巧。